python src/metric/plddt.py \
    --dataset_input /home/tanyang/workspace/diffusion4protein/dataset/Cas9/pdb \
    --out_file cas9_plddt.csv \
    --type protein